// ============================================================================
//
// Copyright (C) 2006-2007 Dengues
//
// Google Group: http://groups.google.com/group/dengues
// QQ Group: 24885404
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
//
// ============================================================================
package org.dengues.designer.ui.database.commands;

import java.util.List;

import org.dengues.designer.ui.editors.AbstractGEFCommand;
import org.dengues.model.database.DBColumn;
import org.dengues.model.database.DBTable;
import org.dengues.model.database.ForeignKey;

/**
 * Qiang.Zhang.Adolf@gmail.com class global comment. Detailled comment <br/>
 * 
 * $Id: Dengues.epf 1 2006-09-29 17:06:40Z qiang.zhang $
 * 
 */
public class DBColumnDeleteCommand extends AbstractGEFCommand {

    private final List<DBColumn> columns;

    private final DBTable table;

    /**
     * Qiang.Zhang.Adolf@gmail.com DBColumnDeleteCommand constructor comment.
     * 
     * @param columns
     * @param table
     */
    public DBColumnDeleteCommand(List<DBColumn> columns, DBTable table) {
        this.columns = columns;
        this.table = table;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.commands.Command#execute()
     */

    /**
     * Qiang.Zhang.Adolf@gmail.com DBColumnDeleteCommand constructor comment.
     * 
     * @param cols
     */
    public DBColumnDeleteCommand(List<DBColumn> cols) {
        columns = cols;
        table = cols.get(0).getTable();
    }

    @Override
    public void execute() {
        for (DBColumn column : columns) {
            table.getColumns().remove(column);
            List<ForeignKey> inputs = column.getInputs();
            List<ForeignKey> outputs = column.getOutputs();
            for (ForeignKey fk : inputs) {
                DBColumn source = fk.getSource();
                source.getOutputs().remove(fk);
            }
            for (ForeignKey fk : outputs) {
                DBColumn target = fk.getTarget();
                target.getInputs().remove(fk);
            }
        }
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.commands.Command#undo()
     */
    @SuppressWarnings("unchecked")//$NON-NLS-1$
    @Override
    public void undo() {
        for (DBColumn column : columns) {
            table.getColumns().add(column);
        }
    }

}
